try:
    import MySQLdb
except ImportError:
    import pymysql

    pymysql.install_as_MySQLdb()